import React from 'react';
import { BasicLayoutProps, Settings as LayoutSettings, MenuDataItem } from '@ant-design/pro-layout';
import { notification } from 'antd';
import { history, RequestConfig } from 'umi';
import RightContent from '@/components/RightContent';
import Footer from '@/components/Footer';
import { ResponseError } from 'umi-request';
import { queryCurrent } from './services/system/user';
import { loginVerify } from './services/system/login';
import defaultSettings from '../config/defaultSettings';
import SysIcon from '@/pages/common/SysIcon';
import { getMenu4User, getMenu4Auth } from '@/services/system/menu';

export async function getInitialState(): Promise<{
  settings?: LayoutSettings;
  currentUser?: API.CurrentUser;
  fetchUserInfo: () => Promise<API.CurrentUser | undefined>;
}> {
  const fetchUserInfo = async () => {
    try {
      const currentUser = await queryCurrent();
      return currentUser;
    } catch (error) {
      history.push('/user/login');
    }
    return undefined;
  };
  // 如果是登录页面，不执行
  if (history.location.pathname !== '/user/login') {
    const currentUser = await fetchUserInfo();
    const menuData = await getMenu4User();
    const menuAuthData = await getMenu4Auth();
    return {
      fetchUserInfo,
      menuData: menuData.result,
      menuAuth: menuAuthData.result,
      currentUser,
      settings: defaultSettings,
    };
  }
  return {
    fetchUserInfo,
    settings: defaultSettings,
    menuData: [],
  };
}

// 加载自定义 Icon
const loopMenuItem = (menus: MenuDataItem[]): MenuDataItem[] => {
  return (
    menus &&
    menus.map(({ icon, children, ...item }) => ({
      ...item,
      access: 'canAdminNot',
      icon: icon && <SysIcon type={icon as string} />,
      children: children && loopMenuItem(children),
    }))
  );
};

export const layout = ({
  initialState,
}: {
  initialState: { settings?: LayoutSettings; currentUser?: API.CurrentUser; menuData?: any[] };
}): BasicLayoutProps => {
  return {
    rightContentRender: () => <RightContent />,
    disableContentMargin: false,
    footerRender: () => <Footer />,
    onPageChange: async () => {
      const { location } = history;
      // history.push('./403');
      try {
        const verify = await loginVerify();
        // const { currentUser } = initialState;
        // 如果没有登录，重定向到 login
        if (verify.result.length === 0 && location.pathname !== '/user/login') {
          history.push('/user/login');
          return;
        }
      } catch (e) {
        console.log(e);
        // if (location.pathname !== '/user/login') {
        //   history.push('/user/login');
        // }
      }
    },
    postMenuData: () => loopMenuItem(initialState && initialState.menuData),
    menuItemRender: (itemProps) => {
      // 左侧菜单，二级子菜单图标隐藏 -> 显示
      return (
        <div onClick={() => history.push(itemProps.path)}>
          {itemProps.icon}
          {itemProps.name}
        </div>
      );
    },
    menuHeaderRender: undefined,
    ...initialState?.settings,
  };
};

const codeMessage = {
  200: '服务器成功返回请求的数据。',
  201: '新建或修改数据成功。',
  202: '一个请求已经进入后台排队（异步任务）。',
  204: '删除数据成功。',
  400: '发出的请求有错误，服务器没有进行新建或修改数据的操作。',
  // 401: '用户没有权限（令牌、用户名、密码错误）。',
  401: '登录超时',
  403: '用户得到授权，但是访问是被禁止的。',
  404: '发出的请求针对的是不存在的记录，服务器没有进行操作。',
  405: '请求方法不被允许。',
  406: '请求的格式不可得。',
  410: '请求的资源被永久删除，且不会再得到的。',
  422: '当创建一个对象时，发生一个验证错误。',
  500: '服务器发生错误，请检查服务器。',
  502: '网关错误。',
  503: '服务不可用，服务器暂时过载或维护。',
  504: '网关超时。',
};

/**
 * 异常处理程序
 */
const errorHandler = (error: ResponseError) => {
  const { response } = error;
  if (response && response.status && response.status !== 401) {
    const errorText = codeMessage[response.status] || response.statusText;
    // const { status, url } = response;

    notification.error({
      message: `请求错误`,
      description: errorText,
    });
  }

  if (!response) {
    notification.error({
      description: '您的网络发生异常，无法连接服务器',
      message: '网络异常',
    });
  }
  throw error;
};

// 设置返回参数
export const request: RequestConfig = {
  errorHandler,
  errorConfig: {
    adaptor: (resData) => {
      return {
        ...resData,
        errorCode: resData.code,
        errorMessage: resData.msg,
      };
    },
  },
  responseInterceptors: [
    (response) => {
      response
        .clone()
        .json()
        .then((data) => {
          if (
            data.code !== '000' &&
            data.code !== '998' &&
            response.url.indexOf('/v2/api-docs') < 0 &&
            response.url.indexOf('/swagger-resources') < 0 &&
            data.code !== undefined
          ) {
            notification.error({
              message: `请求错误:`,
              description: data.msg,
            });
          }
          if (data.code === '998' || data.code === '401') {
            localStorage.removeItem('token');
            history.push('/user/login');
          }
        });
      return response;
    },
  ],
};
